simple python code for beginners